/*
给你一个整数数组 nums 和两个整数 k 和 t 。请你判断是否存在 两个不同下标 i 和 j，使得 abs(nums[i] - nums[j]) <= t ，同时又满足 abs(i - j) <= k 。

如果存在则返回 true，不存在返回 false。

 

示例 1：

输入：nums = [1,2,3,1], k = 3, t = 0
输出：true
示例 2：

输入：nums = [1,0,1,1], k = 1, t = 2
输出：true
示例 3：

输入：nums = [1,5,9,1,5,9], k = 2, t = 3
输出：false
 

提示：

0 <= nums.length <= 2 * 104
-231 <= nums[i] <= 231 - 1
0 <= k <= 104
0 <= t <= 231 - 1

*/

#include "stdc++.h"

// 滑动窗口 + 有序集合
class Solution {
public:
    bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
        int n = nums.size();
        set<int> record;
        for (int i{0}; i < n; ++i) {
            auto iter = record.lower_bound(max(nums[i], INT_MIN + t) - t);
            if (iter != record.end() && *iter <= min(nums[i], INT_MAX - t) + t) {
                return true;
            }
            record.insert(nums[i]);
            if (i >= k) {
                record.erase(nums[i - k]);
            }
        }
        return false;
    }
};

// 桶
class Solution {
public:
    bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
        unordered_map<int, int> mp{};
        int n = nums.size();
        for (int i{0}; i < n; ++i) {
            long x = nums[i];
            int id = getId(x, t + 1ll); // 桶大小 t + 1
            // 同一个桶
            if (mp.count(id)) {
                return true;
            }
            // 相邻桶
            if (mp.count(id - 1) && abs(x - mp[id - 1]) <= t) {
                return true;
            }
            if (mp.count(id + 1) && abs(x - mp[id + 1]) <= t) {
                return true;
            }
            mp[id] = x;
            if (i >= k) {
                mp.erase(getId(nums[i - k], t + 1ll));
            }
        }
        return false;
    }
private:
    int getId(int x, long w) {
        return x < 0 ? (x + 1ll) / w - 1 : x / w;
    }
};